import { BaseAIProvider } from './base';
import type { Model } from '../aiModels';

export class OpenAIProvider extends BaseAIProvider {
  constructor() {
    super(
      'openai',
      'OpenAI',
      'OpenAI API服务，支持GPT-3.5和GPT-4系列模型',
      'https://api.openai.com'
    );
  }

  getDefaultModels(): Model[] {
    return [
      {
        id: 'gpt-3.5-turbo',
        name: 'GPT-3.5 Turbo',
        maxTokens: 16385,
        available: true,
      },
      {
        id: 'gpt-4',
        name: 'GPT-4',
        maxTokens: 8192,
        available: true,
      },
      {
        id: 'gpt-4-turbo',
        name: 'GPT-4 Turbo',
        maxTokens: 128000,
        available: true,
      },
      {
        id: 'gpt-4o',
        name: 'GPT-4o',
        maxTokens: 128000,
        available: true,
      },
    ];
  }
} 